numpy
matplotlib
scipy
torch
torchvision
